import matplotlib.pyplot as plt

from molecular_motion import Molecularmotion

#使用Matplotlib的内置样式
plt.style.use('classic')
#显示中文字体
from matplotlib import rcParams
rcParams['font.family']='SimHei'

#创建一个Molecularmotion实例
mm=Molecularmotion(50_000)
mm.fill_molecular()

#将所有的点都绘制出来
fig,ax=plt.subplots(figsize=(10,6))
point_numbers=range(mm.num_points)
ax.scatter(mm.x_values,mm.y_values,c=point_numbers,cmap=plt.cm.Greens,
            edgecolors='none',s=1)
    
#突出起点和终点
ax.scatter(0,0,c='blue',edgecolors='none',s=100)
ax.scatter(mm.x_values[-1],mm.y_values[-1],c='red',edgecolors='none',
            s=100)
    
#设置图表标题并给坐标轴加上标签
ax.set_title('分子运动',fontsize=24)
  
#隐藏坐标轴
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)

plt.show()
    